Python tip to find square of list elements in 1 line of code